package com.fy56.platform.logistics.utils;

import org.apache.poi.hssf.usermodel.*;
import org.apache.poi.hssf.util.HSSFColor;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.ss.util.CellRangeAddress;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.util.ArrayList;
import java.util.List;

/**
 * [通用工具类]Excel文件操作类
 */
public class ExcelUtil {

	/**
	 * 发送-下载响应流方法
	 * @param response
	 * @param fileName
	 */
	public static void setResponseHeader(HttpServletResponse response, String fileName) {
		try {
			try {
				fileName = new String(fileName.getBytes(),"ISO8859-1");
			} catch (UnsupportedEncodingException e) {
				e.printStackTrace();
			}
			response.setContentType("application/octet-stream;charset=ISO8859-1");
			response.setHeader("Content-Disposition", "attachment;filename="+ fileName);
			response.addHeader("Pargam", "no-cache");
			response.addHeader("Cache-Control", "no-cache");
		} catch (Exception ex) {
			ex.printStackTrace();
		}
	}

	/*----------------------------------------------------------------------------------------------------*/

	/**
	 * [设置]列头单元格样式
	 */
	public static HSSFCellStyle getColumnTopStyle(HSSFWorkbook workbook) {
		  // 设置字体
		  HSSFFont font = workbook.createFont();
		  //设置字体大小
		  font.setFontHeightInPoints((short)11);
		  //字体加粗
		  font.setBoldweight(HSSFFont.BOLDWEIGHT_BOLD);
		  //设置字体名字
		  font.setFontName("Courier New");
		  //设置样式;
		  HSSFCellStyle style = workbook.createCellStyle();
		  //设置底边框;
		  style.setBorderBottom(HSSFCellStyle.BORDER_THIN);
		  //设置底边框颜色;
		  style.setBottomBorderColor(HSSFColor.BLACK.index);
		  //设置左边框;
		  style.setBorderLeft(HSSFCellStyle.BORDER_THIN);
		  //设置左边框颜色;
		  style.setLeftBorderColor(HSSFColor.BLACK.index);
		  //设置右边框;
		  style.setBorderRight(HSSFCellStyle.BORDER_THIN);
		  //设置右边框颜色;
		  style.setRightBorderColor(HSSFColor.BLACK.index);
		  //设置顶边框;
		  style.setBorderTop(HSSFCellStyle.BORDER_THIN);
		  //设置顶边框颜色;
		  style.setTopBorderColor(HSSFColor.BLACK.index);
		  //在样式用应用设置的字体;
		  style.setFont(font);
		  //设置自动换行;
		  style.setWrapText(false);
		  //设置水平对齐的样式为居中对齐;
		  style.setAlignment(HSSFCellStyle.ALIGN_CENTER);
		  //设置垂直对齐的样式为居中对齐;
		  style.setVerticalAlignment(HSSFCellStyle.VERTICAL_CENTER);

		  return style;
	}

	/**
	 * [设置]列数据信息单元格样式
	 */
	public static HSSFCellStyle getStyle(HSSFWorkbook workbook) {
		  // 设置字体
		  HSSFFont font = workbook.createFont();
		  //设置字体大小
		  //font.setFontHeightInPoints((short)10);
		  //字体加粗
		  //font.setBoldweight(HSSFFont.BOLDWEIGHT_BOLD);
		  //设置字体名字
		  font.setFontName("Courier New");
		  //设置样式;
		  HSSFCellStyle style = workbook.createCellStyle();
		  //设置底边框;
		  style.setBorderBottom(HSSFCellStyle.BORDER_THIN);
		  //设置底边框颜色;
		  style.setBottomBorderColor(HSSFColor.BLACK.index);
		  //设置左边框;
		  style.setBorderLeft(HSSFCellStyle.BORDER_THIN);
		  //设置左边框颜色;
		  style.setLeftBorderColor(HSSFColor.BLACK.index);
		  //设置右边框;
		  style.setBorderRight(HSSFCellStyle.BORDER_THIN);
		  //设置右边框颜色;
		  style.setRightBorderColor(HSSFColor.BLACK.index);
		  //设置顶边框;
		  style.setBorderTop(HSSFCellStyle.BORDER_THIN);
		  //设置顶边框颜色;
		  style.setTopBorderColor(HSSFColor.BLACK.index);
		  //在样式用应用设置的字体;
		  style.setFont(font);
		  //设置自动换行;
		  style.setWrapText(false);
		  //设置水平对齐的样式为居中对齐;
		  style.setAlignment(HSSFCellStyle.ALIGN_CENTER);
		  //设置垂直对齐的样式为居中对齐;
		  style.setVerticalAlignment(HSSFCellStyle.VERTICAL_CENTER);

		  return style;
	}

	/*----------------------------------------------------------------------------------------------------*/

	/**
	 * 导出Excel<自定义格式>
	 * @param sheetName sheet名称
	 * @param rowName 标题
	 * @param dataList 内容
	 * @returnHSSFWorkbook对象
	 */
	public static HSSFWorkbook export(String sheetName, List<String> rowName, List<List<String>> dataList){
		HSSFWorkbook workbook = new HSSFWorkbook(); // 创建
		HSSFSheet sheet = workbook.createSheet(sheetName); // 创建工作表

		// 产生表格标题行
		HSSFRow rowm = sheet.createRow(0);
		HSSFCell cellTiltle = rowm.createCell(0);

		// sheet样式定义【getColumnTopStyle()/getStyle()均为自定义方法 - 在下面 - 可扩展】
		HSSFCellStyle columnTopStyle = getColumnTopStyle(workbook);// 获取列头样式对象
		HSSFCellStyle style = getStyle(workbook); // 单元格样式对象

		sheet.addMergedRegion(new CellRangeAddress(0, 1, 0, (rowName.size() - 1)));
		cellTiltle.setCellStyle(columnTopStyle);
		cellTiltle.setCellValue(sheetName);

		// 定义所需列数
		HSSFRow rowRowName = sheet.createRow(2); // 在索引2的位置创建行(最顶端的行开始的第二行)

		// 将列头设置到sheet的单元格中
		for (int n = 0; n < rowName.size(); n++) {
			HSSFCell cellRowName = rowRowName.createCell(n); // 创建列头对应个数的单元格
			cellRowName.setCellType(HSSFCell.CELL_TYPE_STRING); // 设置列头单元格的数据类型
			HSSFRichTextString text = new HSSFRichTextString(rowName.get(n));
			cellRowName.setCellValue(text); // 设置列头单元格的值
			cellRowName.setCellStyle(columnTopStyle); // 设置列头单元格样式
		}

		// 将查询出的数据设置到sheet对应的单元格中
		for (int i = 0; i < dataList.size(); i++) {
			Object[] obj = dataList.get(i).toArray();// 遍历每个对象
			HSSFRow row = sheet.createRow(i + 3);// 创建所需的行数

			for (int j = 0; j < obj.length; j++) {
				HSSFCell cell = null; // 设置单元格的数据类型
				if (j == 0) {
					cell = row.createCell(j, HSSFCell.CELL_TYPE_NUMERIC);
					cell.setCellValue(i + 1);
				} else {
					cell = row.createCell(j, HSSFCell.CELL_TYPE_STRING);
					if (!"".equals(obj[j]) && obj[j] != null) {
						cell.setCellValue(obj[j].toString()); // 设置单元格的值
					}
				}
				cell.setCellStyle(style); // 设置单元格样式
			}
		}
		// 让列宽随着导出的列长自动适应
		for (int colNum = 0; colNum < rowName.size(); colNum++) {
			int columnWidth = sheet.getColumnWidth(colNum) / 256;
			for (int rowNum = 0; rowNum < sheet.getLastRowNum(); rowNum++) {
				HSSFRow currentRow;
				// 当前行未被使用过
				if (sheet.getRow(rowNum) == null) {
					currentRow = sheet.createRow(rowNum);
				} else {
					currentRow = sheet.getRow(rowNum);
				}
				if (currentRow.getCell(colNum) != null) {
					HSSFCell currentCell = currentRow.getCell(colNum);
					if (currentCell.getCellType() == HSSFCell.CELL_TYPE_STRING) {
						int length = currentCell.getStringCellValue().getBytes().length;
						if (columnWidth < length) {
							columnWidth = length;
						}
					}
				}
			}
			if (colNum == 0) {
				sheet.setColumnWidth(colNum, (columnWidth - 2) * 256);
			} else {
				sheet.setColumnWidth(colNum, (columnWidth + 4) * 256);
			}
		}

		return workbook;
	}

	/*----------------------------------------------------------------------------------------------------*/

	/**
	 * 导出Excel<默认格式>
	 * @param sheetName sheet名称
	 * @param rowName 标题
	 * @param dataList 内容
	 * @return HSSFWorkbook对象
	 */
	public static HSSFWorkbook getHSSFWorkbook(String sheetName, List<String> rowName, List<List<String>> dataList){
		// 第一步，创建一个工作簿对象(HSSFWorkbook)，对应一个Excel文件
		HSSFWorkbook workbook = new HSSFWorkbook();

		// 第二步，在workbook中添加一个工作表(sheet),对应Excel文件中的sheet
		HSSFSheet sheet = workbook.createSheet(sheetName);

		// 第三步，在sheet中添加表头第0行,注意老版本poi对Excel的行数列数有限制
		HSSFRow row = sheet.createRow(0);

		// 第四步，创建单元格，并设置值表头 设置表头居中
		HSSFCellStyle style = workbook.createCellStyle();
		style.setAlignment(HSSFCellStyle.ALIGN_CENTER); // 创建一个居中格式

		//声明列对象
		HSSFCell cell = null;

		//创建标题
		for(int i=0;i<rowName.size();i++){
			cell = row.createCell(i);
			cell.setCellValue(rowName.get(i));
			cell.setCellStyle(style);
		}

		//创建内容
		for(int i=0;i<dataList.size();i++){
			row = sheet.createRow(i + 1);
			for(int j=0; j<dataList.get(i).size(); j++){
				//遍历每个对象
				Object[] obj = dataList.get(i).toArray();
				if(obj[j]==null || "".equals(obj[j].toString())){
					obj[j]="";
				}
				//将内容按顺序赋给对应的列对象
				row.createCell(j).setCellValue(obj[j].toString());
			}
		}
		return workbook;
	}

	/*----------------------------------------------------------------------------------------------------*/

	private static final String EXCEL_XLS = "xls";
	private static final String EXCEL_XLSX = "xlsx";

	/**
	 * 判断Excel的版本,获取Workbook
	 * @param in
	 * @param file
	 * @return
	 * @throws IOException
	 */
	public static Workbook getWorkbok(InputStream in, File file) throws IOException{
		Workbook wb = null;
		System.out.printf("file.name = "+file.getName());
		if(file.getName().endsWith(EXCEL_XLS)){  //Excel 2003
			wb = new HSSFWorkbook(in);
		}else if(file.getName().endsWith(EXCEL_XLSX)){// Excel 2007/2010
			wb = new XSSFWorkbook(in);
		}
		return wb;
	}

	/**
	 * 判断文件是否是excel
	 * @throws Exception
	 */
	public static void checkExcelVaild(File file) throws Exception{
		if(!file.exists()){
			throw new Exception("文件不存在");
		}
		if(!(file.isFile() && (file.getName().endsWith(EXCEL_XLS) || file.getName().endsWith(EXCEL_XLSX)))){
			throw new Exception("文件不是Excel");
		}
	}
	
	private static Object getValue(Cell cell) {
		Object obj = null;
		switch (cell.getCellType()) {
			case Cell.CELL_TYPE_BOOLEAN:
				obj = cell.getBooleanCellValue();
				break;
			case Cell.CELL_TYPE_ERROR:
				obj = cell.getErrorCellValue();
				break;
			case Cell.CELL_TYPE_NUMERIC:
				obj = cell.getNumericCellValue();
				break;
			case Cell.CELL_TYPE_STRING:
				obj = cell.getStringCellValue();
				break;
			default:
				break;
		}
		return obj;
	}

	/**
	 * [读取]Excel文件内容<兼容 Excel 2003/2007/2010>
	 * @param path 文件路径
	 * @throws Exception
	 */
	public static List<List<Object>> readXlsFileByPath(String path) {
		List<List<Object>> valList = new ArrayList<>();
		try {
			//[创建]文件对象
			File excelFile = new File(path);
			//[创建]文件流
			FileInputStream in = new FileInputStream(excelFile);

			//[判断]文件是否是excel
			checkExcelVaild(excelFile);
			//[判断]Excel的版本,获取Workbook
			Workbook workbook = getWorkbok(in, excelFile);

			//[获取]表格数量
			int sheetCount = workbook.getNumberOfSheets();// Sheet的数量
			//[判断]表格数量>0
			if(sheetCount>0){
				/**
				 * [设置]当前excel中sheet的下标：从0开始
				 */
				Sheet sheet = workbook.getSheetAt(0);// 遍历第一个Sheet

				//获取总行数
//				System.out.println("数据总行数："+sheet.getLastRowNum());

				// 为跳过第一行目录设置count
				int count = 0;
				for (Row row : sheet) {
					List<Object> val = new ArrayList<>();
					try {
						//跳过第一行的目录
						if(count < 1 ) {
							count++;
							continue;
						}
						//如果当前行没有数据，跳出循环
						if(row.getCell(0).toString().equals("")){
							break;
						}

						//[获取]总列数(空格的不计算)
//						int columnTotalNum = row.getPhysicalNumberOfCells();
//						System.out.println("总列数：" + columnTotalNum + "\t|\t最大列数：" + row.getLastCellNum());

						for (int i = 0; i < row.getLastCellNum(); i++) {
							Cell cell = row.getCell(i);
							if(cell == null) {
								val.add(null);
//								System.out.print("null" + "\t");
							}else{
								//[设置]单元格类型(本文型)
								cell.setCellType(Cell.CELL_TYPE_STRING);
								Object obj = getValue(cell);
								val.add(obj);
//								System.out.print(obj + "\t");
							}
						}
						valList.add(val);
					} catch (Exception e) {
						e.printStackTrace();
					}
				}
			}
			in.close();
		} catch (Exception e) {
			e.printStackTrace();  
		}
		return valList;
	}
	
	/*
	<!-- xls文件读写支持插件 https://mvnrepository.com/artifact/org.apache.poi/poi -->
	<dependency>
		<groupId>org.apache.poi</groupId>
		<artifactId>poi</artifactId>
		<version>3.15</version>
	</dependency>
	<dependency>
		<groupId>org.apache.poi</groupId>
		<artifactId>poi-ooxml</artifactId>
		<version>3.15</version>
	</dependency>

	*/



	/*----------------------------------------------------------------------------------------------------*/
	/*----------------------------------------------临时放置----------------------------------------------*/
	/*----------------------------------------------------------------------------------------------------*/

	/**
	 * 根据指定的地址和文件名,创建文件夹和文件
	 *
	 * @param path     地址
	 * @param fileName 文件名
	 * @param suffix   文件后缀
	 */
	public static File fileEstablish(String path, String fileName, String suffix) {
		File file = new File(path);
		if (!file.exists()) {//判断文件夹是否存在
			file.mkdirs();
		}
		if(fileName!=null && !"".equals(fileName)){
			file = new File(path, fileName + "." + suffix);
		}
		return file;
	}

	/**
	 * 将目标文件或文件夹下,所有文件夹与文件全部删除
	 *
	 * @param file 目标文件/文件夹
	 */
	private static void fileDelete(File file) {
		try {
			//判断文件夹是否存在
			if (file.exists()) {
				//判断是否为文件,直接删除
				if (file.isFile()) {
					file.delete();
				}
				//判断是否为文件夹,先遍历删除里面的文件,最后在把本文件夹删除
				else if (file.isDirectory()) {
					File[] files = file.listFiles();
					for (File fileItem : files) {
						// 递归调用删除可能存在的子文件夹
						fileDelete(fileItem);
					}
					file.delete();
				}
			}
			System.out.println(file + "--->删除成功!");
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	/**
	 * 删除文件夹和文件
	 *
	 * @param path 地址
	 */
	public static void fileDelete(String path) {
		File file = new File(path);
		fileDelete(file);
	}

	/*----------------------------------------------------------------------------------------------------*/
}